import numpy as np
import cmath
import matplotlib.pyplot as plt
import librosa
plt.rcParams['font.sans-serif'] = ['SimHei']  # 用来正常显示中文标签
plt.rcParams['axes.unicode_minus'] = False  # 用来正常显示负号

def subject1():
    kk=np.arange(1,21.0,1)
    ss=np.sum(np.sqrt(kk)*np.exp(120*1j*np.pi*(-0.002)*kk))
    print("ss:",ss,"模以及辐角",cmath.polar(ss))
def assignment2():

    N = 12880
    fs = 8192# 每秒钟的采样率
    n = np.arange(N)
    t = n / fs
    x = 314.5 * np.cos(2 * np.pi * 707 * t - 3.1137) + 527.6 * np.cos(2 * np.pi * 887 * t + 1.3302) + 726.6 * np.cos(
        2 * np.pi * 1170 * t + 1.6029) + 90.24 * np.cos(2 * np.pi * 2116 * t - 0.6233) + 51.36 * np.cos(
        2 * np.pi * 2661 * t + 2.6412) + 39.11 * np.cos(2 * np.pi * 3512 * t + 0.5166)


    librosa.output.write_wav('合成信号.wav',x,fs)#运用系统中的音乐软件，点开这个文件即可听到声音。
    plt.plot(t[1: 1000], x[1: 1000])
    plt.title('合成信号图 ')
    plt.xlabel('时间 /s')
    plt.show()


if __name__ == "__main__":
    assignment2()